# 管理员功能
from core import src

# 添加用户
from interface import admin_interface


def add_user():
    src.register()


# 修改额度
def change_balance():
    while True:
        change_user = input('请输入要修改额度的用户：').strip()
        input_money = input('请输入要修改额度：').strip()

        if not input_money.isdigit():
            print('请重新输入正确的额度！')
            continue
        input_money = int(input_money)
        flag, msg = admin_interface.change_balance_interface(change_user, input_money)
        print(msg)
        if flag:
            break

# 锁定用户
def lock_user():
    while True:
        change_user = input('请输入要锁定的用户：').strip()
        flag, msg = admin_interface.lock_user(change_user)
        print(msg)
        if flag:
            break

# 管理员功能字典
admin_func = {
    '1': add_user,
    '2': change_balance,
    '3': lock_user,
}


def admin_run():
    while True:
        print('''
        1.添加账号
        2.修改额度
        3.冻结账户
        ''')
        choice = input('请输入管理员功能编号：').strip()
        if choice not in admin_func:
            print('请重新输入正确的功能编号！')
            continue
        admin_func[choice]()
